# -*- coding: utf-8 -*-

"""
Created on 2021/6/2 16:35

@File: set_triple_relation.py
@Author: LT
@Description: siteId和objectId可以在4.1步骤中对应每个站点可以获得，可能产生歧义的地方在于objectId和userId2的获得，
              因为站点名不以（租户）、（客户）尾缀区分的话，不知道谁是谁，所以加了尾缀后就好区分了

              ！！！！必须！！！！得是请求体形式，不能用params，我已经用postman试过了！！！
"""
from requests import post
import json
import time


class SetTripleRelation:
    def __init__(self, siteId, objectId, userId1, userId2, **kwargs):
        self.link = 'https://comp.wzies.com/backend/company/site/siteUsersCustomerId'
        self.headers = {'Content-Type': 'application/json'}     # payload形式必须有请求头表示它是什么类型的，这里用json类型！
        self.data = {"siteId": siteId,                          # (可以通过t_site这张数据表根据站点名字往site_name一栏里去查找到)
                     "objectId": objectId,                      # (这个就是数据预览中的objectID)
                     "userId1": userId1,                        # (t_user表中已有的)
                     "userId2": userId2,                        # (t_user表中添通过账号密码新增客户后产生的一个新客户的id),
                     "companyId": "comp_00000001",              # (固定)
                     "userId": "user_00000001",                 # (固定)
                     "timestamp": round(time.time()),           # (当前时间的时间戳)
                     }
        self.res = self.__request()

    def __request(self):
        res = post(url=self.link, data=json.dumps(self.data), headers=self.headers)
        res.close()
        return res


if __name__ == '__main__':
    exp = SetTripleRelation(siteId='site_00009886', objectId='26de40b037804000', userId1='ZJ_016581', userId2='ZJ_016582')
    print(exp.res.json())
    # 返回体:
    # {"code":"1",
    # "msg":"SUCCESS",
    # "data":null,
    # "timestamp":0}
